
import os

def wirte_file():
    fw.write('____________________________________________________________\n')

if __name__ == '__main__':
    WORK_PATH = os.getcwd()  # 获取当前项目工作目录，
    root = os.path.join(WORK_PATH, r'result')
    fp = os.path.join(root, '2020-08-13-' + '站点56080-' + 'LSTM(10years)' + '.txt')
    fw = open(fp, 'a+')
    times = 1
    # fw.write(' test' + str(times) + ' : ' + '\n')
    # fw.write(' epochs :' + str(epochs) + '\n')
    # fw.write(' batch :' + str(n_batch) + '\n')
    # fw.write(' neuron :' + str(n_neurons) + '\n')
    # experiment_two(dataset, epochs, n_batch, n_neurons, test_percent)


    test_percent = 0.9
    n_batch = [10,20,50,100]
    n_neurons =range(120,240,10)
    epochs = [3000,5000,8000]
    times = 1

    for epoch in epochs:
        for batch in n_batch:
            for neurons in n_neurons:
                fw.write(' test ' + str(times) + ' : ' + '\n')
                fw.write(' epochs :' + str(epoch) + '\n')
                fw.write(' batch :' + str(batch) + '\n')
                fw.write(' neuron :' + str(neurons) + '\n')
                times = times + 1
                print('站点56080-LSTM(10years) 本次实验参数： test_percent, n_batch, n_neurons, epochs, optimizer')
                print('站点56080-LSTM(10years) 本次实验参数： ', test_percent, n_batch, n_neurons, epochs)
                wirte_file()
    fw.close()